quadratic equation using python